package top.frsf.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import top.frsf.reggie.common.R;
import top.frsf.reggie.entity.User;
import top.frsf.reggie.service.UserService;
import top.frsf.reggie.utils.SendSms;
import top.frsf.reggie.utils.ValidateCodeUtils;

import javax.servlet.http.HttpSession;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @Author FRSF
 * @Date 2023/1/29 12:56
 * @Description 用户控制层
 */

@Slf4j
@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;
    @Autowired
    private RedisTemplate<Object, Object> redisTemplate;
    @Autowired
    private JavaMailSender javaMailSender;
    @Value("${spring.mail.username}")
    private String sender;

    /**
     * @Author: FRSF
     * @Date: 2023/2/4 22:58
     * @Description 用户登出
     * @param session:
     * @return: top.frsf.reggie.common.R<java.lang.String>
     */
    @PostMapping("/loginout")
    public R<String> loginout(HttpSession session) {
        session.removeAttribute("user");
        return R.error("NOTLOGIN");
    }

    /**
     * @Author: FRSF
     * @Date: 2023/2/4 21:52
     * @Description 发送验证码，后端集成Redis
     * @param user:
     * @return: top.frsf.reggie.common.R<java.lang.String>
     */
    @PostMapping("/sendMsg")
    public R<String> sendMsg(@RequestBody User user) throws Exception {
        // 获取手机号
        if (user.getName().equals("phone")) {
            String phone = user.getPhone();
            // 允许发送的号码
            if ("13086383805".equals(phone)) {
                // 生成随机4位验证码
                String code = ValidateCodeUtils.generateValidateCode(4).toString();
                // 将生成的验证码保存到Redis，并设置有效期为5分钟
                redisTemplate.opsForValue().set(phone, code, 5, TimeUnit.MINUTES);
                log.info("短信发送，号码：{}--验证码：{}", phone, code);

                // 调用阿里云sms API
                SendSms.SendMessage(phone, code);
                return R.success("发送短信成功");
            }
            return R.error("该手机号不在可发送名单内");
        }

        // 获取邮箱
        if (user.getName().equals("email")) {
            String email = user.getPhone();

            // 生成随机4位验证码
            String code = ValidateCodeUtils.generateValidateCode(4).toString();

            SimpleMailMessage mailMessage = new SimpleMailMessage();
            mailMessage.setFrom(sender); //设置发送邮件账号
            mailMessage.setTo(email);    //设置接收邮件的人，可以多个
            mailMessage.setSubject("Reggie邮箱验证");
            mailMessage.setText("您的邮箱验证码为："+ code);
            // 将生成的验证码保存到Redis，并设置有效期为5分钟
            redisTemplate.opsForValue().set(email, code, 5, TimeUnit.MINUTES);
            log.info("短信发送，邮箱：{}--验证码：{}", email, code);

            javaMailSender.send(mailMessage);
            return R.success("发送邮件成功");
        }


        return R.error("验证码发送失败");
    }


    /**
     * @Author: FRSF
     * @Date: 2023/1/29 13:45
     * @Description 前端登录，验证Code
     * @param map: json传的Map需要RequestBody接受
     * @return: top.frsf.reggie.common.R<java.lang.String>
     */
    @PostMapping("/login")
    public R<User> login(@RequestBody Map<String, String> map, HttpSession session) {
        // 获取手机号
        String phone = map.get("phone");
        // 获取验证码
        String code = map.get("code");
        // 从Redis获取保存的验证码，进行验证码校验
        String codeInRedis = (String) redisTemplate.opsForValue().get(phone);


        if (codeInRedis == null) {
            return R.error("不存在该手机/邮箱的验证码或验证码过期");
        }

        if (!codeInRedis.equals(code)) {
            return R.error("验证码错误");
        }

        // 比对成功，说明登录成功
        LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(User::getPhone, phone);
        User user = userService.getOne(wrapper);
        if (user == null) {
            // 若为新用户，需要自动完成注册
            user = new User();
            user.setPhone(phone);
            userService.save(user);
        }
        // 设置userID,过滤器需要检测
        session.setAttribute("user", user.getId());
        // 登录成功，删除Redis中的验证码
        redisTemplate.delete(phone);
        return R.success(user);

    }
}
